Update app.py

app.py

@@ -106,7 +106,7 @@ def resetChat():
     chat_engine.reset()
     return True
 
-def rag(input_text, jezik, btn, file):
+def rag(input_text, history, jezik, file):
 #    if (btn):
 #        resetChat()
     if (file):
@@ -133,42 +133,43 @@ def rag(input_text, jezik, btn, file):
         match jezik:
             case 'hrvatski':
                 o_jezik = 'na hrvatskom jeziku'
-                Settings.llm.system_prompt = system_sr + \
-                    "Call centar telefon je 095 1000 444 za privatne i 095 1000 500 za poslovne korisnike. Stranica podrške je <https://tele mach.hr/podrska>." \
-                    + "Odgovaraj " + o_jezik
+                Settings.llm.system_prompt = system_sr + "Call centar telefon je 095 1000 444 za privatne i 095 1000 500 za poslovne korisnike. Stranica podrške je <https://tele mach.hr/podrska>." + "Odgovaraj " + o_jezik
             case 'slovenski':
                 o_jezik = 'v slovenščini'
-                Settings.llm.system_prompt = system_sr + \
-                    "Call centar i pomoč za fizične uporabnike: 070 700 700.stran za podporo je <https://telemach.si/pomoc>. " \
-                    + "Odgovor " + o_jezik
+                Settings.llm.system_prompt = system_sr + "Call centar i pomoč za fizične uporabnike: 070 700 700.stran za podporo je <https://telemach.si/pomoc>. " + "Odgovor " + o_jezik
             case 'srpski':
                 o_jezik = 'na srpskom jeziku'
-                Settings.llm.system_prompt = system_sr + \
-                    "Call centar telefon je 19900 za sve korisnike. Stranica podrške je <https://sbb.rs/podrska/>. " \
-                    + "Odgovaraj " + o_jezik
+                Settings.llm.system_prompt = system_sr + "Call centar telefon je 19900 za sve korisnike. Stranica podrške je <https://sbb.rs/podrska/>. " + "Odgovaraj " + o_jezik
             case 'makedonski':
                 o_jezik = 'на македонски јазикот'
-                Settings.llm.system_prompt = system_sr + \
-                    "Stranica podrške je https://mn.nettvplus.com/me/podrska/ za NetTV. " \
-                    + "Oдговори " + o_jezik
+                Settings.llm.system_prompt = system_sr + "Stranica podrške je https://mn.nettvplus.com/me/podrska/ za NetTV. " + "Oдговори " + o_jezik
             case 'Eksperimentalna opcija':
                 o_jezik = 'N/A'
-                Settings.llm.system_prompt = system_sr + \
-                    "Call centar telefon je 12755 za Crnu Goru, 0800 31111 za BIH, 070 700 700 u Sloveniji, 19900 u Srbiji, 095 1000 444 za hrvatske korisnike. "
+                Settings.llm.system_prompt = system_sr + "Call centar telefon je 12755 za Crnu Goru, 0800 31111 za BIH, 070 700 700 u Sloveniji, 19900 u Srbiji, 095 1000 444 za hrvatske korisnike. "
 #        if (o_jezik!='N/A'):
 #            input_text += " - odgovori " + o_jezik + "."
 #        return query_engine.query(input_text)
         return chat_engine.chat(input_text)
-
-iface = gr.Interface(fn=rag, inputs=[gr.Textbox(label="Pitanje:", lines=6),
-                                     gr.Dropdown(["slovenski", "hrvatski", "srpski", "makedonski", "Eksperimentalna opcija"], value="srpski", label="Jezik", info="N/A"),
-                                     gr.Button("Resetuj razgovor", variant='secondary'),
-                                     gr.File(),
-                                    ], 
-                     outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
-                     title="Kako Vam mogu pomoći?",
-                     description="UChat"
-                    )
+# Interface
+# gr.Textbox(label="Pitanje:", lines=6),
+# outputs=[gr.Textbox(label="Odgovor:", lines=6)], 
+with gr.Blocks as iface:
+    j = gr.Dropdown(["slovenski", "hrvatski", "srpski", "makedonski", "Eksperimentalna opcija"], value="srpski", label="Jezik", info="N/A")
+    f = gr.File()
+    gr.ChatInterface(rag,        
+        title="Kako Vam mogu pomoći?",
+        description="UChat",
+        chatbot=gr.Chatbot(height=300),
+        textbox=gr.Textbox(placeholder="Pitanje ili opis problema", container=False, scale=7),
+        description="Postavite pitanje ili opišite problem koji imate",
+        theme="soft",
+        examples=["Ne radi mi internet", "Koje usluge imam na raspologanju?", "Ne radi mi daljinski upravljač, šta da radim?"],
+        cache_examples=True,
+        retry_btn=None,
+        undo_btn="Briši prethodno",
+        clear_btn="Clear",
+        additional_inputs 0 = [j, f]
+    )
 
 #with gr.Blocks() as iface:
 #    gr.Markdown("Uchat")