charantejapolavarapu commited on
Commit
a6372fd
·
verified ·
1 Parent(s): 490b948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +97 -64
app.py CHANGED
@@ -1,70 +1,103 @@
1
  import gradio as gr
2
- from huggingface_hub import InferenceClient
3
-
4
-
5
- def respond(
6
- message,
7
- history: list[dict[str, str]],
8
- system_message,
9
- max_tokens,
10
- temperature,
11
- top_p,
12
- hf_token: gr.OAuthToken,
13
- ):
14
- """
15
- For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
16
- """
17
- client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
18
-
19
- messages = [{"role": "system", "content": system_message}]
20
-
21
- messages.extend(history)
22
-
23
- messages.append({"role": "user", "content": message})
24
-
25
- response = ""
26
-
27
- for message in client.chat_completion(
28
- messages,
29
- max_tokens=max_tokens,
30
- stream=True,
31
- temperature=temperature,
32
- top_p=top_p,
33
- ):
34
- choices = message.choices
35
- token = ""
36
- if len(choices) and choices[0].delta.content:
37
- token = choices[0].delta.content
38
-
39
- response += token
40
- yield response
41
-
42
-
43
- """
44
- For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
45
- """
46
- chatbot = gr.ChatInterface(
47
- respond,
48
- type="messages",
49
- additional_inputs=[
50
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
51
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
52
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
53
- gr.Slider(
54
- minimum=0.1,
55
- maximum=1.0,
56
- value=0.95,
57
- step=0.05,
58
- label="Top-p (nucleus sampling)",
59
- ),
60
- ],
61
  )
62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
  with gr.Blocks() as demo:
64
- with gr.Sidebar():
65
- gr.LoginButton()
66
- chatbot.render()
67
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
68
 
69
- if __name__ == "__main__":
70
- demo.launch()
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
+ import random
4
+
5
+ # -------------------------------
6
+ # Load Models
7
+ # -------------------------------
8
+
9
+ # Speech to Text Model (Whisper)
10
+ asr = pipeline(
11
+ "automatic-speech-recognition",
12
+ model="openai/whisper-base"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  )
14
 
15
+ # Text Generation Model (LLM)
16
+ generator = pipeline(
17
+ "text-generation",
18
+ model="mistralai/Mistral-7B-Instruct-v0.1",
19
+ device_map="auto"
20
+ )
21
+
22
+ # -------------------------------
23
+ # Question Bank
24
+ # -------------------------------
25
+
26
+ questions = [
27
+ "Explain overfitting in machine learning.",
28
+ "What is the difference between supervised and unsupervised learning?",
29
+ "Explain gradient descent in simple terms.",
30
+ "What is the role of activation functions in neural networks?",
31
+ "What is the difference between CNN and RNN?"
32
+ ]
33
+
34
+ # -------------------------------
35
+ # Interview Logic
36
+ # -------------------------------
37
+
38
+ def start_interview():
39
+ question = random.choice(questions)
40
+ return question
41
+
42
+ def evaluate_answer(audio, question):
43
+
44
+ # Convert Speech to Text
45
+ result = asr(audio)
46
+ user_answer = result["text"]
47
+
48
+ # Create evaluation prompt
49
+ prompt = f"""
50
+ You are a technical interviewer.
51
+
52
+ Question: {question}
53
+ Candidate Answer: {user_answer}
54
+
55
+ Evaluate the answer and give:
56
+ 1. Technical Accuracy Score (out of 10)
57
+ 2. Clarity Score (out of 10)
58
+ 3. Overall Score (out of 10)
59
+ 4. Improvement Suggestions
60
+
61
+ Keep feedback concise.
62
+ """
63
+
64
+ # Generate Evaluation
65
+ output = generator(
66
+ prompt,
67
+ max_new_tokens=300,
68
+ do_sample=True,
69
+ temperature=0.7
70
+ )
71
+
72
+ feedback = output[0]["generated_text"]
73
+
74
+ return f"📝 Transcribed Answer:\n{user_answer}\n\n📊 Evaluation:\n{feedback}"
75
+
76
+
77
+ # -------------------------------
78
+ # Gradio UI
79
+ # -------------------------------
80
+
81
  with gr.Blocks() as demo:
 
 
 
82
 
83
+ gr.Markdown("# 🎤 Smart Interview Simulator (AI Voice Bot)")
84
+ gr.Markdown("Answer the question using your voice.")
85
+
86
+ question_output = gr.Textbox(label="Interview Question")
87
+
88
+ start_button = gr.Button("Start Interview")
89
+ start_button.click(start_interview, outputs=question_output)
90
+
91
+ audio_input = gr.Audio(source="microphone", type="filepath")
92
+
93
+ submit_button = gr.Button("Submit Answer")
94
+
95
+ result_output = gr.Textbox(label="Evaluation Feedback")
96
+
97
+ submit_button.click(
98
+ evaluate_answer,
99
+ inputs=[audio_input, question_output],
100
+ outputs=result_output
101
+ )
102
 
103
+ demo.launch()