fix for UI

app.py

@@ -5,6 +5,8 @@ from helpers.model_utils import get_model
 from aspect_and_sentiment_extraction import extract_aspects_and_sentiment
 from answer_bot import answer_question
 
+transcript_file_name = None
+
 def summarize(transcript_file_name):
     chunked_docs = load_rtf_document_and_chunk(transcript_file_name)
 
@@ -15,7 +17,11 @@ def extract_aspects(transcript_file_name):
     # Implement your aspect extraction and sentiment analysis logic here
     return extract_aspects_and_sentiment(transcript_file_name)
 
+
 def get_answer_for(user_question):
+    if transcript_file_name is None:
+        return "No Transcript Uploaded, Upload RTF File First", ""
+    
     # Answer the user's question using the question-answering model
     if user_question.strip():  # Ensure there is a question provided
         answer_text = answer_question(question=user_question)
@@ -25,9 +31,8 @@ def get_answer_for(user_question):
     return answer_text.lstrip()
 
 def process_transcript(uploaded_file):
-    # Process the file
-    transcript_file_name = uploaded_file.name
-    print(f"Transcript File Name :{transcript_file_name}")
+    if transcript_file_name is None:
+        return "No Transcript Uploaded, Upload RTF File First", ""
     
     # Summarize the content
     summary = summarize(transcript_file_name=transcript_file_name).lstrip()
@@ -37,15 +42,23 @@ def process_transcript(uploaded_file):
 
     return summary, sentiment
 
+def setup_rtf_file_handle(uploaded_file):
+    transcript_file_name = uploaded_file.name
+    print(f"Transcript File Name :{transcript_file_name}")
+
 with gr.Blocks() as demo:
-    with gr.Tab("Aspect/Sentiment"):
+    with gr.Group("Upload RTF File"):
         rtf_file = gr.File(label="Podcast Transcript RTF file")
+        submit_button = gr.Button("Upload File")
+        submit_button.click(setup_rtf_file_handle)
+    with gr.Group("Aspects and Sentiment of Podcast"):
         summary = gr.Textbox(label="Summary of Podcast")
         sentiment = gr.Textbox(label="Aspect Based Sentiments")
-        submit_button = gr.Button("Submit")
-        submit_button.click(process_transcript, inputs=rtf_file, outputs=[summary, sentiment])
-    
-    with gr.Tab("Question/Answer"):
+        submit_button = gr.Button("Generate Aspects and Summary")
+        submit_button.click(process_transcript, inputs=rtf_file, outputs=[summary, sentiment])    
+
+    with gr.Group("Question/Answer"):
+        gr.Markdown("Question/Answer")
         question = gr.Textbox(label="Question")
         answer = gr.Textbox(label="Answer")
         answer_button = gr.Button("Answer Question")