Update app.py

app.py

@@ -85,32 +85,32 @@ def caption_question_interface(image):
         words = caption.split()
         # Generate questions for the first word
         if len(words) > 0:
-            keyword = words[0]
-            question = generate_questions(caption, keyword)
-            questions_with_keywords.extend([(q, keyword) for q in question])
+            answer = words[0]
+            question = generate_questions(caption, answer)
+            questions_with_answers.extend([(q, answer) for q in question])
         # Generate questions for the second word
         if len(words) > 1:
-            keyword = words[1]
-            question = generate_questions(caption, keyword)
-            questions_with_keywords.extend([(q, keyword) for q in question])
+            answer = words[1]
+            question = generate_questions(caption, answer)
+            questions_with_answers.extend([(q, answer) for q in question])
         # Generate questions for the second word + first word
         if len(words) > 1:
-            keyword = " ".join(words[:2])
-            question = generate_questions(caption, keyword)
-            questions_with_keywords.extend([(q, keyword) for q in question])
+            answer = " ".join(words[:2])
+            question = generate_questions(caption, answer)
+            questions_with_answers.extend([(q, answer) for q in question])
         # Generate questions for the third word
         if len(words) > 2:
-            keyword = words[2]
-            question = generate_questions(caption, keyword)
-            questions_with_keywords.extend([(q, keyword) for q in question])
+            answer = words[2]
+            question = generate_questions(caption, answer)
+            questions_with_answers.extend([(q, answer) for q in question])
         # Generate questions for the fourth word
         if len(words) > 3:
-            keyword = words[3]
-            question = generate_questions(caption, keyword)
-            questions_with_answers.extend([(q, keyword) for q in question])
+            answer = words[3]
+            question = generate_questions(caption, answer)
+            questions_with_answers.extend([(q, answer) for q in question])
 
     # Format questions with answers
-    formatted_questions = [f"Question: {q}\nKeyword: {a}" for q, a in questions_with_keywords]
+    formatted_questions = [f"Question: {q}\nKeyword: {a}" for q, a in questions_with_answers]
     formatted_questions = "\n".join(formatted_questions)
 
     # Return the generated captions and formatted questions with answers
@@ -124,7 +124,7 @@ gr_interface = gr.Interface(
         gr.Textbox(label="Generated Questions")
     ],
     title="Visual Question Generator",
-    description="Generate captions and questions for images using pre-trained models.",
+    description="Generate captions and questions for images using Arabic image captioning model and question generation model",
     theme=seafoam,
 )