File size: 5,289 Bytes
7ce19a5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3dd0c02
 
7ce19a5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3dd0c02
7ce19a5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3dd0c02
7ce19a5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3dd0c02
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
'use client'

import { useState } from 'react'
import {
  Container,
  Paper,
  Typography,
  TextField,
  Button,
  Box,
  Alert,
  CircularProgress,
  Chip,
  Divider,
  Stack
} from '@mui/material'
import MedicalInformationIcon from '@mui/icons-material/MedicalInformation'
import HealthAndSafetyIcon from '@mui/icons-material/HealthAndSafety'
import LocalHospitalIcon from '@mui/icons-material/LocalHospital'

const exampleSymptoms = [
  'Headache',
  'Fever',
  'Cough',
  'Fatigue',
  'Nausea'
]

const exampleHistory = [
  'Hypertension',
  'Diabetes',
  'Asthma',
  'Allergies'
]

export default function Home() {
  const [symptoms, setSymptoms] = useState('')
  const [medicalHistory, setMedicalHistory] = useState('')
  const [recommendation, setRecommendation] = useState('')
  const [loading, setLoading] = useState(false)

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault()
    setLoading(true)

    try {
      const response = await fetch('/api/recommend', {
        method: 'POST',
        headers: {
          'Content-Type': 'application/json',
        },
        body: JSON.stringify({ symptoms, medicalHistory }),
      })

      const data = await response.json()
      setRecommendation(data.recommendation)
    } catch (error) {
      console.error('Failed to get recommendation:', error)
    } finally {
      setLoading(false)
    }
  }

  const addExampleSymptom = (symptom: string) => {
    setSymptoms(prev => prev ? `${prev}, ${symptom}` : symptom)
  }

  const addExampleHistory = (history: string) => {
    setMedicalHistory(prev => prev ? `${prev}, ${history}` : history)
  }

  return (
    <Container maxWidth="md" sx={{ py: 4 }}>
      <Paper elevation={3} sx={{ p: 4, borderRadius: 2 }}>
        <Box display="flex" alignItems="center" gap={2} mb={4}>
          <LocalHospitalIcon color="primary" sx={{ fontSize: 40 }} />
          <Typography variant="h4" component="h1" fontWeight="bold">
            AI Drug Recommendation System
          </Typography>
        </Box>

        <Alert severity="info" sx={{ mb: 4 }}>
          This is an AI-powered tool to assist healthcare professionals. All recommendations should be reviewed by a qualified medical practitioner.
        </Alert>

        <form onSubmit={handleSubmit}>
          <Stack spacing={4}>
            <Box>
              <Typography variant="h6" display="flex" alignItems="center" gap={1} mb={2}>
                <MedicalInformationIcon />
                Symptoms
              </Typography>
              <TextField
                fullWidth
                multiline
                rows={4}
                value={symptoms}
                onChange={(e) => setSymptoms(e.target.value)}
                placeholder="Enter patient symptoms..."
                required
              />
              <Box mt={2}>
                <Typography variant="subtitle2" mb={1}>Example symptoms:</Typography>
                <Stack direction="row" spacing={1} flexWrap="wrap" gap={1}>
                  {exampleSymptoms.map((symptom) => (
                    <Chip
                      key={symptom}
                      label={symptom}
                      onClick={() => addExampleSymptom(symptom)}
                      clickable
                    />
                  ))}
                </Stack>
              </Box>
            </Box>

            <Box>
              <Typography variant="h6" display="flex" alignItems="center" gap={1} mb={2}>
                <HealthAndSafetyIcon />
                Medical History
              </Typography>
              <TextField
                fullWidth
                multiline
                rows={4}
                value={medicalHistory}
                onChange={(e) => setMedicalHistory(e.target.value)}
                placeholder="Enter patient medical history..."
                required
              />
              <Box mt={2}>
                <Typography variant="subtitle2" mb={1}>Example conditions:</Typography>
                <Stack direction="row" spacing={1} flexWrap="wrap" gap={1}>
                  {exampleHistory.map((history) => (
                    <Chip
                      key={history}
                      label={history}
                      onClick={() => addExampleHistory(history)}
                      clickable
                    />
                  ))}
                </Stack>
              </Box>
            </Box>

            <Button
              type="submit"
              variant="contained"
              size="large"
              disabled={loading}
              startIcon={loading ? <CircularProgress size={20} /> : null}
            >
              {loading ? 'Generating Recommendation...' : 'Get Recommendation'}
            </Button>
          </Stack>
        </form>

        {recommendation && (
          <Box mt={4}>
            <Divider sx={{ my: 4 }} />
            <Typography variant="h6" gutterBottom>
              Recommendation
            </Typography>
            <Paper variant="outlined" sx={{ p: 3, bgcolor: 'background.default' }}>
              <Typography whiteSpace="pre-wrap">
                {recommendation}
              </Typography>
            </Paper>
          </Box>
        )}
      </Paper>
    </Container>
  )
}