Upload app.py

app.py

@@ -22,7 +22,7 @@ Please carefully read the following text. The text is written in {LANG} language
 {TEXT}  
 </text>
 
-After reading it, I want you to extract topic informations from text.
+After reading it, I want you to extract topic informations from text in {LANG} language.
 '''
 
 template_json = '''
@@ -33,7 +33,7 @@ Your task is to read the following text, convert it to json format using 'Answer
 
 Your final response MUST contain only the response, no other text.
 Example:
-{{"Answer":"Sport, Entertainment"}}
+{{"Answer":["Sport", "Entertainment", "General", "Inflation", "Effects of Inflation"]]}}
 '''
 json_output_parser = JsonOutputParser()
 
@@ -44,6 +44,14 @@ def classify_text(text):
     start = time.time()
     lang = detect(text)
 
+    language_map = {"tr": "turkish",
+                    "en": "english",
+                    "ar": "arabic",
+                    "es": "spanish",
+                    "it": "italian",
+                    }
+    lang = language_map[lang]
+
     prompt_classify = PromptTemplate(
         template=template_classify,
         input_variables=["LANG", "TEXT"]
@@ -72,7 +80,7 @@ def gradio_app(text):
 def create_gradio_interface():
     with gr.Blocks() as iface:
         text_input = gr.Textbox(label="Text")
-        output_text = gr.Textbox(label="Classify")
+        output_text = gr.Textbox(label="Topics")
         time_taken = gr.Textbox(label="Time Taken (seconds)")
         submit_btn = gr.Button("Classify")