code changes

app.py

@@ -1,7 +1,112 @@
 import gradio as gr
+from transformers import pipeline
+
+import re
+
+email_subject_pipeline = pipeline("text-generation", model="kkasiviswanath/distilgpt2_email_subject_summarizer")
+
+def clean_subject(response):
+    print(response)
+    lst = response.split('<|sep|>')
+    if (len(lst) >= 2):
+        response = lst[1].replace("<|endoftext|>","")
+    return response
+
+
+def generate_subject(email: str):
+    prompt = f"<|startoftext|> {email} <|sep|>"
+
+    # Use the pipeline to generate the text
+    sample_outputs = email_subject_pipeline(prompt, max_new_tokens=12, num_beams=5, early_stopping=True, num_return_sequences=1)
+
+    # The pipeline output is a list of dictionaries, so extract the generated text
+    subject = clean_subject(sample_outputs[0]['generated_text'])
+    return subject
+
+
 
 def greet(name):
     return "Hello " + name + "!!"
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+theme = gr.themes.ThemeClass.from_hub("freddyaboulton/dracula_revamped")
+
+with gr.Blocks(theme=theme) as demo:
+    gr.Markdown("# AI-based Generative QA System")
+    with gr.Tab("Email Subject Line Generation"):
+        gr.Markdown("To generate subject line for your email, copy/paste or type email in the box and click on Generate Subject")
+        with gr.Row():
+            with gr.Column(scale=2):
+                text_input_email = gr.Textbox(lines=8, label="Email")
+                text_output_subject = gr.Textbox(label="Generated Subject")
+            with gr.Column(scale=1):
+                with gr.Column(scale=1):
+                    gr.Markdown(
+                        """
+                        ## Team 17
+                        ```
+
+                        Archit Garg
+                        Dinesh
+                        Chandrasekhar B
+                        K Kasi Viswanath
+                        ```
+                        ## Model
+                        ```
+                        Fine tuned gpt2-medium
+
+                        ```
+                        ## Training Dataset
+                        ```
+                        AESLC
+                        ```
+
+                        """)
+        with gr.Row():
+            with gr.Column(scale=2):
+                btn_generate_subject = gr.Button("Generate Subject")
+            with gr.Column():
+                gr.Markdown(
+                    """
+                    """)
+    with gr.Tab("Question Answering on AIML Queries"):
+        with gr.Column(scale=1):
+            gr.Markdown("To generate answer for your question, copy/paste or type question in the box and click on Generate Answer")
+            with gr.Row():
+                with gr.Column(scale=2):
+                    text_input_question = gr.Textbox(label="Question")
+                    text_output_answer = gr.Textbox(lines=8, label="Generated Answer")
+                with gr.Column(scale=1):
+                    with gr.Column(scale=1):
+                        gr.Markdown(
+                            """
+                            ## Team 17
+                            ```
+
+                           Archit Garg
+                           Dinesh
+                           Chandrasekhar B
+                           K Kasi Viswanath
+                            ```
+                            ## Model
+                            ```
+                            Fine tuned gpt2-medium
+
+                            ```
+                            ## Training Dataset
+                            ```
+                            Custom
+                            ```
+
+                            """)
+            with gr.Row():
+                with gr.Column(scale=2):
+                    btn_generated_answer = gr.Button("Generate Answer")
+                with gr.Column():
+                    gr.Markdown(
+                        """
+                        """)
+    btn_generate_subject.click(generate_subject, inputs=text_input_email, outputs=text_output_subject)
+
+    btn_generated_answer.click(greet, inputs="text", outputs="text")
+    
+demo.launch(share=True)

requirements.txt

@@ -0,0 +1,2 @@
+torch
+transformers